import os
import logging

padding = {'width': 512, 'height': 512}
is_pad = True

num_classes = 1
num_stacks = 1
num_blocks = 4

lr = 1e-6
batch_size = 1

"""
neckline_left,neckline_right,
center_front,
shoulder_left,shoulder_right,
armpit_left,armpit_right,
waistline_left,waistline_right,
cuff_left_in,cuff_left_out,cuff_right_in,cuff_right_out,
top_hem_left,top_hem_right,
waistband_left,waistband_right,
hemline_left,hemline_right,
crotch,
bottom_left_in,bottom_left_out,bottom_right_in,bottom_right_out 
"""
all2cloth ={ 'Blouse':  [1, 1, 1, 1, 1, 1, 1,    1, 1, 1, 1,     1, 1, 0, 0,     0, 0, 0, 0,    0, 0, 0, 0, 0],
            'outwear':  [1, 1, 0, 1, 1, 1, 1,    1, 1, 1, 1,     1, 1, 1, 1,     0, 0, 0, 0,    0, 0, 0, 0, 0],
           'trousers':  [0, 0, 0, 0, 0, 0, 0,    0, 0, 0, 0,     1, 1, 0, 0,     1, 1, 1, 1,    1, 1, 0, 0, 1],
              'dress':  [1, 1, 1, 1, 1, 1, 1,    1, 1, 1, 1,     0, 0, 0, 0,     1, 1, 0, 0,    0, 0, 1, 1, 0],
              'skirt':  [0, 0, 0, 0, 0, 0, 0,    0, 0, 0, 0,     0, 0, 0, 0,     1, 1, 1, 1,    0, 0, 0, 0, 0]}


data_dir = '/data/Datasets/guanls/ccf'
log_dir = '/data/Datasets/songmj/ccf/logdir_Deepfashion'
model_dir = '/data/Datasets/songmj/projects/Deepfashion'
prepared_data_dir = "/data/Datasets/songmj/projects/Deepfashion/"

log_level = 'info'
model_path = os.path.join(model_dir, 'latest')
save_steps = 1000

num_workers = 4
num_GPU = 1
device_id = 2

logger = logging.getLogger('train')
logger.setLevel(logging.INFO)

fh = logging.FileHandler(os.path.join(log_dir, 'train.log'))
fh.setLevel(logging.INFO)
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)

formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
logger.addHandler(fh)
logger.addHandler(ch)
